package com.bailian.common.core.handler;

import org.apache.ibatis.type.BaseTypeHandler;
import org.apache.ibatis.type.JdbcType;
import org.apache.ibatis.type.MappedTypes;
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONException;

import java.sql.CallableStatement;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

/**
 * MyBatis JSON类型处理器
 * 用于处理数据库JSON字段与Java String之间的转换
 * 
 * @author bailian
 * @date 2025-01-27
 */
@MappedTypes(String.class)
public class JsonTypeHandler extends BaseTypeHandler<String> {

    @Override
    public void setNonNullParameter(PreparedStatement ps, int i, String parameter, JdbcType jdbcType) throws SQLException {
        if (parameter != null) {
            ps.setString(i, parameter);
        } else {
            ps.setNull(i, jdbcType.TYPE_CODE);
        }
    }

    @Override
    public String getNullableResult(ResultSet rs, String columnName) throws SQLException {
        String value = rs.getString(columnName);
        return parseJson(value);
    }

    @Override
    public String getNullableResult(ResultSet rs, int columnIndex) throws SQLException {
        String value = rs.getString(columnIndex);
        return parseJson(value);
    }

    @Override
    public String getNullableResult(CallableStatement cs, int columnIndex) throws SQLException {
        String value = cs.getString(columnIndex);
        return parseJson(value);
    }

    /**
     * 解析JSON字符串
     * 
     * @param value 数据库中的值
     * @return 解析后的字符串
     */
    private String parseJson(String value) {
        if (value == null || value.trim().isEmpty()) {
            return null;
        }
        
        try {
            // 如果是有效的JSON格式，直接返回
            // 如果不是JSON格式，也作为普通字符串返回
            return value;
        } catch (JSONException e) {
            // 如果解析失败，返回原始字符串
            return value;
        }
    }
} 